Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add metadata quantized models in PTQ & GPTQ facades. #1034

Merged
merged 11 commits into from
Apr 11, 2024
Merged

Add metadata quantized models in PTQ & GPTQ facades. #1034

merged 11 commits into from
Apr 11, 2024

Conversation

elad-c
Copy link
Collaborator

@elad-c elad-c commented Apr 8, 2024

Pull Request Description:

This PR adds the MCTQ metadata feature to MCT quantized models.
Enabled from the TPC. Added imx500.v2 TPCs that enable metadata.

Checklist before requesting a review:

  • I set the appropriate labels on the pull request.
  • I have added/updated the release note draft (if necessary).
  • I have updated the documentation to reflect my changes (if necessary).
  • All function and files are well documented.
  • All function and classes have type hints.
  • There is a licenses in all file.
  • The function and variable names are informative.
  • I have checked for code duplications.
  • I have added new unittest (if necessary).

@elad-c
Add metadata quantized models in PTQ & GPTQ facades. Enabled from the…
17a4078 
… TPC.

Added imx500.v2 TPCs that enable metadata.
@elad-c elad-c requested review from ofirgo and reuvenperetz April 8, 2024 08:53
ofirgo
ofirgo requested changes Apr 8, 2024
View reviewed changes
model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_keras_exporter.py Outdated Show resolved Hide resolved
model_compression_toolkit/exporter/model_exporter/pytorch/fakely_quant_onnx_pytorch_exporter.py Show resolved Hide resolved
model_compression_toolkit/exporter/model_exporter/pytorch/fakely_quant_onnx_pytorch_exporter.py Show resolved Hide resolved
...atform_capabilities/target_platform/targetplatform2framework/target_platform_capabilities.py Outdated Show resolved Hide resolved
...atform_capabilities/target_platform/targetplatform2framework/target_platform_capabilities.py Outdated Show resolved Hide resolved
model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2/tpc_keras.py Show resolved Hide resolved
@elad-c elad-c merged commit e295990 into main Apr 11, 2024
27 checks passed
@elad-c elad-c deleted the add_metadata branch April 11, 2024 12:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@reuvenperetz reuvenperetz reuvenperetz approved these changes

@ofirgo ofirgo ofirgo approved these changes

Assignees
No one assigned
Labels
auto:core auto:exporter auto:gptq auto:ptq auto:target_platform_capabilities auto:tests enhancement New feature or request keras pr: minor feature pytorch
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@elad-c @ofirgo @reuvenperetz